1740D - Knowledge Cards - CodeForces Solution

#include <bits/stdc++.h>
#define ll long long
#define FOR(i, n) for(ll i = 0; i < n; i++)
#define D 8

using namespace std;


template<typename T>
ostream& operator<<(ostream &out, vector<T> cont)
{
    for(auto it = cont.begin(); it != cont.end(); it++)
        out << *it << (it != cont.end()-1 ? ' ' : '\n');
    return out;
}

template<typename T>
ostream& operator<<(ostream &out, vector<vector<T>> cont)
{
    for(auto it = cont.begin(); it != cont.end(); it++)
        out << *it;
    return out;
}


int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);

    ll t;
    cin >> t;

    FOR(i, t)
    {
        ll n, m, k, j=0;
        cin >> n >> m >> k;
        set<ll> s;
        string res = "ya\n";
        vector<ll> v(k);
        FOR(i, k) cin >> v[i];

        for(ll i = k; i>0; i--)
        {
            while (s.count(i) == 0)
            {
                s.insert(v[j++]);
                if(s.size() > n*m-3) 
                {
                    res = "tidak\n";
                }
            }
            s.erase(i);           
        }
        cout << res;
    }

    return 0;
}